Interactive demo of image recognition

In this demo, we present a basic demonstration of our project by training the qGAN to recognize an 8 $\times$ 8 pixel art of smiley face. Enjoy :)

Module import

First, install python modules as following:

import numpy as np
import plotly.io as pio
pio.renderers.default = "notebook"

Second, import our self-written module in the --imgen.py--

from src.imgen import ImageGenerator

Image loading

Now, initialze the input pixel art.

NUM_QUBITS = 6
NUM_LAYERS = 2

EPOCH_SAMPLE_SIZE = 10**4
BATCH_SAMPLE_SIZE = 10**3

pixel_art = np.array([
    [0, 0, 0, 0, 0, 0, 0, 0],
    [0, 0, 1, 0, 0, 1, 0, 0],
    [0, 0, 1, 0, 0, 1, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0],
    [0, 1, 0, 0, 0, 0, 1, 0],
    [0, 1, 0, 0, 0, 0, 1, 0],
    [0, 0, 1, 1, 1, 1, 0, 0],
    [0, 0, 0, 0, 0, 0, 0, 0]
], dtype=np.float)

Initialize the qGAN class for image recognition

imgen = ImageGenerator(
    NUM_QUBITS, NUM_LAYERS,
    epoch_sample_size=EPOCH_SAMPLE_SIZE, batch_sample_size=BATCH_SAMPLE_SIZE,
    enable_remapping=True
)

Load the pixel art. After preprocessing with Gaussian filters, this is what your original art looks like!

imgen.load_image(pixel_art, blur_sigma=0.6, show_figure=True)

qGAN training

Train the qGAN and see how the output of the generator changes in 300 epochs! (Spoiler alert: you can see your smiley face again after only ~30 epochs!)

NUM_EPOCHS = 300

imgen.train(imgen.make_dataset(), NUM_EPOCHS, show_progress=True)

Training epoch 300 of 300:

Generate interactive probability distribution output with training epoch

imgen.plot_output_distribution_history()

Plot loss function versus steps. A trend towards convergence can be seen.

imgen.plot_loss_history()

Plot the final averaged distribution output. It matches pretty well with the original pixel face!

imgen.plot_final_output_distribution(avg_window=5)

Get the final averaged variational parameters for the circuit

imgen.get_final_params(avg_window=10)

array([[[-0.91242063, -0.6248723 , -0.21710499, -0.13538274,
         -0.06022143, -0.00766536],
        [ 0.01763456, -0.11306028,  0.09204257, -0.02819966,
         -0.02732457, -0.01349152]]], dtype=float32)